

class Param(object):
    def __init__(self, params=None):
        super(Param, self).__init__()
        if params is not None:
            self.data_path = params['data_path']
            self.file_name = params['file_name']
        self.data_path = './data'
        self.file_name = 'tang.npz'
        self.batch_size = 128

        # LSTMModel
        self.vocab_size = 8293
        self.embedding_dim = 64
        self.hidden_dim = 128
        self.num_layer = 2

        # train
        self.epoch_num = 100
        self.lr = 1e-3
        self.device = 'cpu'
        self.save_path = './outputs'
        self.load_model = True
        self.model_path = './outputs'
